package main

import (
	"fmt"
	"net"
	"strings"
	"time"
)

// 创建用户结构体类型
type Client struct {
	C    chan string
	Name string
	Addr string
}

// 创建全局map，创建在线用户
var onlineMap map[string]Client

// 创建全局channel 传递用户消息
var message = make(chan string)

func WriteMsgToClient(clnt Client, conn net.Conn) {
	// 监听用户自带 channel 上是否有消息
	for msg := range clnt.C {
		conn.Write([]byte(msg + "\n"))
	}

}

func MakeMsg(clnt Client, msg string) (buf string) {
	buf = "[" + clnt.Addr + "]" + clnt.Name + ": " + msg
	return
}
func HandlerConnect(conn net.Conn) {
	defer conn.Close()
	// 创建channel 判断用户是否活跃
	hasData := make(chan bool)
	// 获取用户网络IP+port
	netAddr := conn.RemoteAddr().String()
	// 创建新连接用户结构体信息
	clnt := Client{
		C:    make(chan string),
		Name: netAddr,
		Addr: netAddr,
	}
	// 将新连接用户添加至在线用户map中 key: ip+port, value: Client
	onlineMap[netAddr] = clnt

	// 创建专门用来给当前用户发送消息的go程
	go WriteMsgToClient(clnt, conn)
	// 发送用户上线消息到 message 通道中
	// message <- "[" + netAddr + "]" + clnt.Name + "login"
	message <- MakeMsg(clnt, "login")

	// 创建一个channel ，用来判断用户退出状态
	isQuit := make(chan bool)

	buf := make([]byte, 4096)
	go func() {
		for {
			n, err := conn.Read(buf)
			if n == 0 { // 从网络socket读取完成
				isQuit <- true

				fmt.Printf("检测到%s客户端推出\n", clnt.Name)
				return
			}
			if err != nil {
				fmt.Println("conn.Read err", err)
				return
			}
			// 将读到的用户消息广播给所有在线用户, 写入到 message
			msg := string(buf[:n-1]) // n-1 因为客户端发送诸如 exit的字符串会加上 \n n-1表示去除\n
			// 提取用户在线列表
			if msg == "who" && len(msg) == 3 {
				conn.Write([]byte("online user list:\n"))
				// 遍历当前 map，获取在线用户列表
				for _, user := range onlineMap {
					userInfo := user.Addr + ":" + user.Name + "\n"
					conn.Write([]byte(userInfo))
				}
				// 判断用户发送了 改名 命令
			} else if len(msg) >= 8 && msg[:6] == "rename" {
				newName := strings.Split(msg, "|")[1]
				clnt.Name = newName
				onlineMap[netAddr] = clnt // 更新在线用户列表 onlineMap
				conn.Write([]byte("rename sucessful\n"))
			} else {
				message <- MakeMsg(clnt, msg)
			}
			hasData <- true
		}
	}()

	// 创建一个匿名go程，专门处理用户发送的消息。

	for {
		// 监听channel 上的数据流动
		select {
		case <-isQuit:
			close(clnt.C)                      // 父go程结束后，子go程不会随之结束，需要手动关闭
			delete(onlineMap, clnt.Addr)       // 将用户从在线用户列表移除
			message <- MakeMsg(clnt, "logout") // 写入用户退出消息到全局channel
			return
		case <-hasData:
			// 什么都不做，目的是重置下面case 计时器
		case <-time.After(time.Second * 10): // 超时强制退出
			close(clnt.C)
			delete(onlineMap, clnt.Addr)       // 将用户从在线用户列表移除
			message <- MakeMsg(clnt, "logout") // 写入用户退出消息到全局channel
			return
		}
	}

}

func Manager() {
	// 初始化 onlineMap
	onlineMap = make(map[string]Client)
	//监听全局channel中是否有数据, 有数据存储到msg，无数据阻塞
	for {

		msg := <-message

		// 循环发送消息给所有在线用户
		for _, clnt := range onlineMap {
			clnt.C <- msg
		}
	}

}

func main() {
	// 创建套接字
	listener, err := net.Listen("tcp", "127.0.0.1:8000")
	if err != nil {
		fmt.Println("Listen err", err)
		return
	}

	defer listener.Close()
	// 创建管理者go程，管理map和全局channel
	go Manager()

	// 循环监听套接字连接请求
	for {
		conn, err := listener.Accept()
		if err != nil {
			fmt.Println("listener.Accept err", err)
			return
		}
		// 启动go程处理客户端数据请求
		go HandlerConnect(conn)
	}
}
